"""
公共方法
"""

from hashlib import md5


def encrypting_pwd(password: str) -> str:
    """使用MD5加密密码"""
    md5_obj = md5()
    md5_obj.update(password.encode('utf-8'))
    # 加盐
    salt = '加多了，有点咸。。。'
    md5_obj.update(salt.encode('utf-8'))

    return md5_obj.hexdigest()


def login_auth(func):
    """验证登录状态的装饰器"""
    from core import src

    def wrapper(*args, **kwargs):
        if src.login_user:
            res = func(*args, **kwargs)
            return res
        else:
            print('当前未登录，请登陆后使用！')
            src.login()
            
    return wrapper
